package com.example.healthcondition.security;

import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import io.jsonwebtoken.security.Keys;
import org.springframework.stereotype.Component;

import java.util.Date;

@Component
public class JwtTokenUtil {

    private final String secretKey = "dGhpc2lzdGhlc2FmZXJrZXlmb3Jqd3RzYXV0aG9yaXphdGlvbg==";;  // 使用实际的密钥
    private final long expirationTime = 86400000; // 设置过期时间（24小时）

    // 解析 token 并返回用户名
    public String parseToken(String token) {
        Claims claims = getClaimsFromToken(token);
        return claims.getSubject();  // 返回 token 中存储的用户名
    }

    // 获取 token 中的 userId
    public Long getUserIdFromToken(String token) {
        Claims claims = getClaimsFromToken(token);
        return claims.get("userId", Long.class);  // 返回 token 中存储的 userId
    }

    // 检查 token 是否过期
    public boolean isTokenExpired(String token) {
        return getClaimsFromToken(token).getExpiration().before(new Date());
    }

    // 获取 token 中的 Claims
    private Claims getClaimsFromToken(String token) {
        return Jwts.parser()
                .setSigningKey(secretKey)
                .parseClaimsJws(token)
                .getBody();
    }

    // 生成 JWT
    public String generateToken(String username, Integer userId) {
        return Jwts.builder()
                .setSubject(username)
                .claim("userId", userId)  // 添加 userId 作为自定义 Claim
                .setIssuedAt(new Date())
                .setExpiration(new Date(System.currentTimeMillis() + expirationTime))
                .signWith(SignatureAlgorithm.HS256, secretKey)
                .compact();
    }
}
